import os


def ensure_directories(path,user_id):
    # 获取项目根目录
    # 检查 data 目录是否存在
    if not os.path.exists(path):
        os.makedirs(path)  # 创建 data 目录
        print(f"创建目录: {path}")

    # 检查用户 ID 目录是否存在
    user_dir = os.path.join(path, str(user_id))
    if not os.path.exists(user_dir):
        os.makedirs(user_dir)  # 创建用户目录
        print(f"创建目录: {user_dir}")
    else:
        print(f"目录已存在: {user_dir}")

def get_filenames(directory):
    if not os.path.exists(directory):
        print(f"目录不存在: {directory}")
        return []

    return [f for f in os.listdir(directory) if f.endswith(".zip") and os.path.isfile(os.path.join(directory, f))]